package cn.bugstack.chatgpt.data.types.enums;

import lombok.AllArgsConstructor;
import lombok.Getter;

@Getter
@AllArgsConstructor
public enum ChatGLMModel {

    // 文本对话模型
    GLM_3_5_TURBO("glm-3-turbo"),
    GLM_4_PLUS("glm-4-plus"),//高端旗舰，性能对标chatgpt4o
    GLM_4("glm-4"),

    GLM_4V_PLUS_0111("glm-4v-plus-0111"),//25年最新更新，手动加入，多模态模型

    //免费，0成本大模型
    GLM_4_Flash("glm-4-flash"),

    //最新图像生成模型
    COGVIEW_4("cogview-4")
    ;

    private final String code;

    public static ChatGLMModel get(String code){
        switch (code){
            case "glm-4":
                return ChatGLMModel.GLM_4;
            case "glm-4-plus":
                return ChatGLMModel.GLM_4_PLUS;
            case "glm-3-turbo":
                return ChatGLMModel.GLM_3_5_TURBO;
            case "glm-4v-plus-0111":
                return ChatGLMModel.GLM_4V_PLUS_0111;
            case "glm-4-flash":
                return ChatGLMModel.GLM_4_Flash;
            case "cogview-4":
                return ChatGLMModel.COGVIEW_4;
            default:
                return ChatGLMModel.GLM_3_5_TURBO;
        }
    }

}
